import openpyxl
from openpyxl import Workbook,load_workbook
import os



#实例化Workbook对象
wb = load_workbook('数据集/学号.xlsx')
#打开wb中的表'Sheet1'
ws = wb['Sheet1']
ws.title = 'DataBase'
#获取行列数
row = ws.max_row
column = ws.max_column
#初始化数组存储ws中的值
database = [[0 for c in range(column)] for r in range(row)]
for i in range(0,row):
    for j in range(0,column):
        #由于python数组从0开始，ws的rc从1开始，故+1
        database[i][j] = ws.cell(row = i+1,column = j+1).value
#步骤1完成



dir = '数据集/新建文件夹'
#os.walk()会用递归的方式遍历路径下的所有文件夹
#root为当前路径，dirs为当前路径下子目录，files为当前路径下非目录文件
#此处仅需将files作为对象进行保存
file = []
for root,dirs,files in os.walk(dir):
    file.extend (files)
# 分离文件名与后缀
testdata = []
ex = []
#os.path.splitext()会将其拆分并返回两个字符串
for i in range(len(file)):
    (filename,extension) = os.path.splitext(file[i])
    testdata.append(filename)
    ex.append(extension)
#步骤2完成



n = len(database)
m = len(testdata)
i=0
j=0
while j!=m:
    if database[i][0] == testdata[j]:
        #将数据表汇总对应的第二列数据存为mark
        ws.cell(i+1,2).value = 'mark'
        i += 1
        j += 1
    else:
        i += 1
#将修改后的数据表储存到原文件中
wb.save('数据集/学号.xlsx')
#步骤3完成